package com.heima.wemedia.service.impl;

import com.alibaba.fastjson.JSON;
import com.heima.aliyun.scan.GreenScan;
import com.heima.aliyun.scan.ScanResult;
import com.heima.common.constants.message.PublishArticleConstants;
import com.heima.common.constants.wemedia.WemediaConstants;
import com.heima.common.exception.CustException;
import com.heima.feigns.AdminFeign;
import com.heima.model.common.dtos.ResponseResult;
import com.heima.model.common.enums.AppHttpCodeEnum;
import com.heima.model.wemedia.pojos.WmNews;
import com.heima.utils.common.SensitiveWordUtil;
import com.heima.wemedia.mapper.WmNewsMapper;
import com.heima.wemedia.service.WmNewsAutoScanService;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.lang3.StringUtils;
import org.springframework.amqp.rabbit.core.RabbitTemplate;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.stereotype.Service;

import java.util.*;
import java.util.stream.Collectors;


@Service
@Slf4j
public class WmNewsAutoScanServiceImpl implements WmNewsAutoScanService {
    @Autowired
    private WmNewsMapper wmNewsMapper;

    @Value("${file.oss.web-site}")
    private String webSite;

    @Autowired
    GreenScan greenScan;

    @Autowired
    AdminFeign adminFeign;

    @Autowired
    RabbitTemplate rabbitTemplate;

    /**
     * 文章自动审核功能
     * @param id  自媒体文章id
     */
    @Override
    public void autoScanWmNews(Integer id) {

        // 1. 判断文章id是否为空
        if(id == null){
            CustException.cust(AppHttpCodeEnum.PARAM_INVALID,"待审核的文章id为空");
        }

        //2.根据id查询文章
        WmNews wmNews = wmNewsMapper.selectById(id);
        if (wmNews == null) {
            CustException.cust(AppHttpCodeEnum.DATA_NOT_EXIST,"对应的文章不存在");
        }
        // 3. 判断文章的状态  必须是 1 (待审核)  (避免重复消费)
        Short status = wmNews.getStatus();
        if(!WemediaConstants.WM_NEWS_SUMMIT_STATUS.equals(status)){
            log.info("当前文章状态 为: {}   不是待审核状态，无需审核",status);
            return;
        }

        //4.抽取文章中所有的文本和图片
        Map<String,Object> contentAndImages = handleTextAndImages(wmNews);
        String text = contentAndImages.get("text").toString();
        List<String> image = (List<String>) contentAndImages.get("image");

        //5.利用DFA算法工具类，进行自定义敏感词过滤
        Boolean aBoolean = handleSensitive(text, wmNews);
        if(!aBoolean){
            // 返回false代表审核不成功   结束处理
            log.info("文章审核未通过，原因: 内容中包含敏感词");
            return;
        }

        //6.阿里云文本审核
        boolean scanText = handleTextScan(text,wmNews);
        if(!scanText){
            // 返回false代表审核不成功   结束处理
            log.info("文章审核未通过，原因: 内容中包含违规词汇");
            return;
        }

        // 7. 阿里云的图片审核  2 有违规图片    3 不确定/aliyun未调用成功   通过继续下一步
        boolean scanImages = handleImageScan(image,wmNews);
        if(!scanImages){
            // 返回false代表审核不成功   结束处理
            log.info("文章审核未通过，原因: 图片中包含违规信息");
            return;
        }

        // 8. 将文章状态改为 8 自动审核通过
        updateWmNews(WmNews.Status.SUCCESS.getCode(), "审核通过",wmNews);

        // 9. TODO 根据文章的发布时间，发送延迟消息 用于定时发布文章( 9 已发布)
        // 通知定时发布文章
        // 发布时间
        long publishTime = wmNews.getPublishTime().getTime();
        // 当前时间
        long nowTime = new Date().getTime();
        long remainTime = publishTime - nowTime;
        // 发布文章
        rabbitTemplate.convertAndSend(PublishArticleConstants.DELAY_DIRECT_EXCHANGE
                , PublishArticleConstants.PUBLISH_ARTICLE_ROUTE_KEY
                , wmNews.getId()
                , (message)->{                              // 延时消息 必设置
                    message.getMessageProperties().setHeader("x-delay",remainTime<=0?0:remainTime);
                    return message;
                }
        );
        log.info("立即发布文章通知成功发送，文章id : {}", wmNews.getId());


    }


    /**
     * 阿里云检测图片
     * @param images
     * @param wmNews
     * @return
     */
    private boolean handleImageScan(List<String> images, WmNews wmNews) {
        boolean flag = false;
        try {
            ScanResult scanResult = greenScan.imageUrlScan(images);
            switch (scanResult.getSuggestion()){
                case "block":
                    // 失败
                    updateWmNews(WmNews.Status.FAIL.getCode(), "文章图片中违规信息: "+scanResult.getLabel() ,wmNews);
                    break;
                case "review":
                    // 人工
                    updateWmNews(WmNews.Status.ADMIN_AUTH.getCode(), "文章图片中有不确定因素，需要进一步人工审核" ,wmNews);
                    break;
                case "pass":
                    // 成功
                    flag = true;
                    break;
                default:
                    // 人工
                    updateWmNews(WmNews.Status.ADMIN_AUTH.getCode(), "阿里云调用状态异常，需要进一步人工审核:" ,wmNews);
                    break;
            }
        } catch (Exception e) {
            e.printStackTrace();
            // 人工审核
            updateWmNews(WmNews.Status.ADMIN_AUTH.getCode(), "阿里云调用失败，需要进一步人工审核:" ,wmNews);
        }
        return flag;

    }

    /**
     * 阿里云检测文本
     * @param text
     * @param wmNews
     * @return
     */
    private boolean handleTextScan(String text, WmNews wmNews) {
        boolean flag = false;
        try {
            ScanResult result = greenScan.greenTextScan(text);
            String suggestion = result.getSuggestion();
            switch (suggestion){
                case "block":
                    // 失败
                    updateWmNews(WmNews.Status.FAIL.getCode(), "文章内容中包含违规词汇: "+result.getLabel() ,wmNews);
                    break;
                case "review":
                    // 人工
                    updateWmNews(WmNews.Status.ADMIN_AUTH.getCode(), "文章内容中有不确定因素，需要进一步人工审核:" ,wmNews);
                    break;
                case "pass":
                    // 成功
                    flag = true;
                    break;
                default:
                    // 人工
                    updateWmNews(WmNews.Status.ADMIN_AUTH.getCode(), "阿里云调用状态异常，需要进一步人工审核:" ,wmNews);
                    break;
            }
        } catch (Exception e) {
            e.printStackTrace();
            // 人工
            updateWmNews(WmNews.Status.ADMIN_AUTH.getCode(), "阿里云调用失败，需要进一步人工审核:" ,wmNews);
        }

        return flag;

    }

    /**
     * DFA检测文本
     * @param text
     * @param wmNews
     * @return
     */
    private Boolean handleSensitive(String text, WmNews wmNews) {
        Boolean flag = true;

        //远程调用admin微服务，查询全部敏感词
        ResponseResult<List<String>> sensitives = adminFeign.sensitives();
        //判断远程调用查询是否成功
        if (!sensitives.checkCode()) {
            CustException.cust(AppHttpCodeEnum.REMOTE_SERVER_ERROR);
        }
        List<String> sensitivesData = sensitives.getData();

        SensitiveWordUtil.initMap(sensitivesData);

        Map<String, Integer> map = SensitiveWordUtil.matchWords(text);

        if (map.size()>0) {
            updateWmNews(WmNews.Status.FAIL.getCode(), "文章中包含敏感词: "+map ,wmNews);
            flag = false;
        }

        return flag;

    }

    private void updateWmNews(Short status,String reason,WmNews wmNews) {
        wmNews.setStatus(status);
        wmNews.setReason(reason);
        wmNewsMapper.updateById(wmNews);
    }

    /**
     * 抽取文章中所有的文本和图片
     * @param wmNews
     * @return
     */
    private Map<String, Object> handleTextAndImages(WmNews wmNews) {

        //创建一个Map集合用于保存文本和图片信息
        HashMap<String, Object> map = new HashMap<>();


        String content = wmNews.getContent();
        //内容信息反序列化成list<Map>集合
        List<Map> list = JSON.parseArray(content, Map.class);

        //获取所有的文本内容(内容文本+标题文本)
        String contentText = list.stream()
                .filter(x -> x.get("type").equals("text"))
                .map(x -> x.get("value").toString())
                .collect(Collectors.joining("_hmtt_"));
        String title = wmNews.getTitle();

        contentText= contentText+"_hmtt_"+title;


        //获取所有的图片内容（内容图片+封面图片）
        List<String> collectImage = list.stream()
                .filter(x -> x.get("type").equals("image"))
                .map(x -> x.get("value").toString())
                .distinct()
                .collect(Collectors.toList());

        String images = wmNews.getImages();
        if (StringUtils.isNotBlank(images)) {
            String[] split = images.split(",");

            List<String> fengmian = Arrays.stream(split)
                    .map(x -> webSite + x)
                    .collect(Collectors.toList());
            //合并
            collectImage.addAll(fengmian);
        }

        //去重
        collectImage = collectImage.stream().distinct().collect(Collectors.toList());
        //保存到Map集合
        map.put("text",contentText);
        map.put("image",collectImage);

        return map;

    }
}
